# 使用 NVIDIA 官方的 CUDA 基础镜像
FROM nvidia/cuda:12.2.0-base-ubuntu22.04

# 设置环境变量，避免在安装过程中出现交互提示
ENV DEBIAN_FRONTEND=noninteractive

# 更新包列表并安装常见依赖
RUN apt-get update && apt-get upgrade -y && \
    apt-get install -y \
        build-essential \
        curl \
        wget \
        git \
        vim \
        sudo \
        unzip \
        zip \
        supervisor\
        net-tools \
        software-properties-common \
        gnupg2 \
        && rm -rf /var/lib/apt/lists/*

# 安装常用编程语言和工具（可根据需要添加）
RUN apt-get update && apt-get install -y \
        python3 \
        python3-pip \
        && rm -rf /var/lib/apt/lists/*

# 安装 Node.js（示例）
RUN curl -fsSL https://deb.nodesource.com/setup_20.x | bash - && \
    apt-get install -y nodejs && \
    rm -rf /var/lib/apt/lists/*

# 安装 NVIDIA 实用工具（如 nvidia-smi）
RUN apt-get update && apt-get install -y \
        nvidia-utils-535 \
        && rm -rf /var/lib/apt/lists/*

# 设置工作目录
WORKDIR /workspace

RUN wget https://github.com/get-convex/convex-backend/releases/download/precompiled-2025-01-08-9fe39c9/convex-local-backend-x86_64-unknown-linux-gnu.zip
RUN unzip convex-local-backend-x86_64-unknown-linux-gnu.zip
RUN curl -fsSL https://ollama.com/install.sh | sh

RUN mkdir /workspace/ai-town-cn-mount

# 设置默认命令（可根据需要修改）
CMD [ "bash" ]